python3 hls4ml_pred.py --load_model mnist_bayes_lenet --num_bayes_layer 1 --num_mc_samples 2 --gpus 0,1 --dataset mnist